import requests
import json
import logging

# 配置日志
logging.basicConfig(
    level=logging.INFO,  # 设置日志级别为 INFO
    format="%(asctime)s - %(levelname)s - %(message)s",  # 设置日志格式
    handlers=[
        logging.FileHandler("test5_cli.log"),  # 将日志写入文件
        logging.StreamHandler()  # 同时将日志输出到控制台
    ]
)



# 服务端地址
SERVER_URL = "http://127.0.0.1:5000/get_embeddings"

def main():
    # 定义要发送的文本
    texts = ["今天天气很好", "我们一起去郊游吧"]

    # 构造请求数据
    data = {"texts": texts}

    try:
        # 发送 POST 请求
        logging.info('发送请求...')
        response = requests.post(SERVER_URL, json=data)

        # 检查响应状态码
        if response.status_code == 200:
            # 解析响应数据
            result = response.json()
            logging.info("收到的嵌入向量数据:")
            logging.info(json.dumps(result, indent=4, ensure_ascii=False))
        else:
            print(f"请求失败，状态码: {response.status_code}")
            print(f"错误信息: {response.text}")

    except Exception as e:
        print(f"请求过程中发生错误: {str(e)}")

if __name__ == "__main__":
    main()